﻿// Copyright 2007-2018 Chris Patterson, Dru Sellers, Travis Smith, et. al.
//  
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
// this file except in compliance with the License. You may obtain a copy of the 
// License at 
// 
//     http://www.apache.org/licenses/LICENSE-2.0 
// 
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 
// CONDITIONS OF ANY KIND, either express or implied. See the License for the 
// specific language governing permissions and limitations under the License.
namespace MassTransit.Serialization
{
    using System;
    using System.Net.Mime;
    using System.Runtime.Serialization;
    using GreenPipes;
    using Newtonsoft.Json;
    using Newtonsoft.Json.Bson;
    using Util;


    public class EncryptedMessageDeserializerV2 :
        IMessageDeserializer
    {
        readonly ICryptoStreamProviderV2 _cryptoStreamProvider;
        readonly JsonSerializer _deserializer;
        readonly IObjectTypeDeserializer _objectTypeDeserializer;

        public EncryptedMessageDeserializerV2(JsonSerializer deserializer, ICryptoStreamProviderV2 cryptoStreamProvider)
        {
            _deserializer = deserializer;
            _cryptoStreamProvider = cryptoStreamProvider;
            _objectTypeDeserializer = new ObjectTypeDeserializer(_deserializer);
        }

        void IProbeSite.Probe(ProbeContext context)
        {
            var scope = context.CreateScope("encrypted");
            scope.Add("contentType", ContentType.MediaType);
            _cryptoStreamProvider.Probe(scope);
        }

        public ContentType ContentType => EncryptedMessageSerializerV2.EncryptedContentType;

        public ConsumeContext Deserialize(ReceiveContext receiveContext)
        {
            try
            {
                using (var body = receiveContext.GetBodyStream())
                {
                    using (var disposingCryptoStream = _cryptoStreamProvider.GetDecryptStream(body, receiveContext))
                    {
                        using (var jsonReader = new BsonDataReader(disposingCryptoStream))
                        {
                            var messageEnvelope = _deserializer.Deserialize<MessageEnvelope>(jsonReader);

                            return new JsonConsumeContext(_deserializer, _objectTypeDeserializer, receiveContext,
                                messageEnvelope);
                        }
                    }
                }
            }
            catch (JsonSerializationException ex)
            {
                throw new SerializationException(
                    "A JSON serialization exception occurred while deserializing the message envelope", ex);
            }
            catch (SerializationException)
            {
                throw;
            }
            catch (Exception ex)
            {
                throw new SerializationException("An exception occurred while deserializing the message envelope", ex);
            }
        }
    }
}